package com.acwing.www;

import java.util.Arrays;

/**
 * Creared with IntelliJ IDEA.
 * Description:堆排序 时间复杂度 O(n*logn) 空间复杂度O（1） 稳定性 不稳定
 * User:yxd
 * Date:2022-01-24
 * Time:16:07
 */
public class Heap {
    public static void main(String[] args) {
        int[] array = {2,4,1,23,15,24,55,14};
        long start = System.currentTimeMillis();//开始时间
        heapSort(array);
        long end = System.currentTimeMillis();//结束时间
        System.out.println(end - start);
        System.out.println(Arrays.toString(array));
    }
    public static void heapSort(int[] array){
        createHeap(array);//建堆 O(N)
        int end = array.length - 1;
        while(end > 0){//交换队首队尾的元素，来保证有序O(N*logn)
            int tmp = array[0];
            array[0] = array[end];
            array[end] = tmp;
            shiftDown(array,0,end);
            end--;
        }
    }
    public static void createHeap(int[] array){
        for (int parent = (array.length - 1 - 1) / 2; parent >= 0 ; parent--) {
            shiftDown(array,parent,array.length);//调整每一棵树
        }
    }

    /**
     * 调整元素的位置保证是一个完整的堆
     * @param array 数组
     * @param parent 父亲节点
     * @param len 要调整的距离
     */
    public static void shiftDown(int[] array,int parent,int len){
        int child = 2 * parent + 1;
        while(child < len){
            if(child + 1 < len && array[child + 1] > array[child]){
                child ++;
            }
            if(array[child] > array[parent]){
                int tmp = array[parent];
                array[parent] = array[child];
                array[child] = tmp;
                parent = child;
                child = 2 * parent  + 1;
            }else{
                break;
            }

        }
    }
}
